from langdev_helper.llm.qwen import llm0

from langchain.chains import ConversationChain
from langchain.memory import ConversationBufferMemory
from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder

memory = ConversationBufferMemory(return_messages=True)
chain = ConversationChain(llm=llm0, memory=memory)

def qa(q):
    print('\nuser: ', q)
    res = chain.invoke({"input": q})
    print('AI:', res['response'])

qa("你好，我的名字是张三，我是一名程序员。")
qa("南京是哪个省？")
qa("我告诉过你我的名字，是什么？，我的职业是什么？")

